Micron Document
--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
| SparkN0de-git | SparkN0de |
--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------


Displaying Raw • Download


tests/backend/conftest.py 64e4fceb2d8acf83803b97a87ea8168489fcd0ec (64e4fceb) Text, 1.62 KB

Tff7b72import T7ee787asyncio
Tff7b72import T7ee787os
Tff7b72import T7ee787tempfile
Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3patch

Tff7b72import T7ee787pytest

T8b949e# Set log dir to a temporary directory for tests to avoid permission issues
T8b949e# in restricted environments like sandboxes.
Te6edf3osTff7b72.Td2a8ffenvironTb4b4b4[Ta5d6ff"Ta5d6ffMESHCHAT_LOG_DIRTa5d6ff"Tb4b4b4] Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)


Tf0883e@pytestTff7b72.Td2a8fffixtureTb4b4b4(Te6edf3autouseTff7b72=Tff7b72TrueTb4b4b4)
Tff7b72def Td2a8ffglobal_mocksTb4b4b4(Tb4b4b4)Tb4b4b4:
Tff7b72with Tb4b4b4(
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.AsyncUtilsTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_async_utilsTb4b4b4,
Te6edf3patchTb4b4b4(
Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.IdentityContext.start_background_threadsTa5d6ff"Tb4b4b4,
Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4,
Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.generate_ssl_certificateTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffasyncio.sleepTa5d6ff"Tb4b4b4, Te6edf3side_effectTff7b72=Tff7b72lambda Tff7b72*Te6edf3argsTb4b4b4, Tff7b72*Tff7b72*Te6edf3kwargsTb4b4b4: Te6edf3asyncioTff7b72.Td2a8ffsleepTb4b4b4(T79c0ff0Tb4b4b4)Tb4b4b4)Tb4b4b4,
Tb4b4b4)Tb4b4b4:
T8b949e# Mock run_async to properly close coroutines
Tff7b72def Td2a8ffmock_run_asyncTb4b4b4(Te6edf3coroTb4b4b4)Tb4b4b4:
Tff7b72if Te6edf3asyncioTff7b72.Td2a8ffiscoroutineTb4b4b4(Te6edf3coroTb4b4b4)Tb4b4b4:
Tff7b72tryTb4b4b4:
T8b949e# If it's a coroutine, we should close it if it's not being awaited
Te6edf3coroTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)
Tff7b72except Tf85149RuntimeErrorTb4b4b4:
Tff7b72pass
Tff7b72elif Tffa657hasattrTb4b4b4(Te6edf3coroTb4b4b4, Ta5d6ff"Ta5d6ff__await__Ta5d6ff"Tb4b4b4)Tb4b4b4:
T8b949e# Handle other awaitables
Tff7b72pass

Te6edf3mock_async_utilsTff7b72.Td2a8ffrun_asyncTff7b72.Td2a8ffside_effect Tff7b72= Te6edf3mock_run_async

Tff7b72yield Tb4b4b4{
Ta5d6ff"Ta5d6ffasync_utilsTa5d6ff"Tb4b4b4: Te6edf3mock_async_utilsTb4b4b4,
Tb4b4b4}


Tf0883e@pytestTff7b72.Td2a8fffixtureTb4b4b4(Te6edf3autouseTff7b72=Tff7b72TrueTb4b4b4)
Tff7b72def Td2a8ffcleanup_sqlite_connectionsTb4b4b4(Tb4b4b4)Tb4b4b4:
Tff7b72yield
T8b949e# After each test, try to close any lingering sqlite connections if possible
T8b949e# This is a bit hard globally without tracking them, but we can at least
T8b949e# trigger GC which often helps with ResourceWarnings.
Tff7b72import T7ee787gc

Te6edf3gcTff7b72.Td2a8ffcollectTb4b4b4(Tb4b4b4)


──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────